
# import model
# import Trainer
# import os
# if __name__ == '__main__':
#     net = model.PNet()
#     if not os.path.exists("./param"):
#         os.makedirs("./param")
#     trainer = Trainer.Trainer(net, './param/p_net.pth', r"E:\CelebA\MTCN\dataSet\12")
#     trainer.trainer(0.01)

# import model
# import Trainer
# import os
# if __name__ == '__main__':
#     net = model.RNet()
#     if not os.path.exists("./param"):
#         os.makedirs("./param")
#     trainer = Trainer.Trainer(net, './param/r_net.pth', r"E:\CelebA\MTCN\dataSet\24")
#     trainer.trainer(0.001)

import model
import Trainer
import os
if __name__ == '__main__':
    net = model.ONet()
    if not os.path.exists("./param"):
        os.makedirs("./param")
    trainer = Trainer.Trainer(net, './param/o_net.pth', r"E:\CelebA\MTCN\dataSet\48")
    trainer.trainer(0.0003)